# 将图像全部按照文件夹分类
# 00001D_ADF.jpg  -> ADF/00001D.png


import os,shutil
# 这个是源图像文件
dir = r"D:\cycleFusion-project\dataset-1444\fusedmodels\VIFB-113\fused_images"
# 下面这个是目标文件夹的上一级
short_dir = "D:\cycleFusion-project\msrs-113\VIFP"

# 读取图片
files = os.listdir(dir)
print("在拷贝：")
# 进行重命名
for f in files:
    name,nearest_dir = f.split('.')[0].split('_',maxsplit=1)         # 00001D   ADF

    long_dir = os.path.join(short_dir,nearest_dir)        # 要保存的新目录
    os.makedirs(long_dir,exist_ok=True)

    path = os.path.join(dir, f)         # 原路径
    target_path = os.path.join(long_dir, name + ".png")     # 新路径

    shutil.copyfile(path, target_path)  # 移动
    print(f,end = ",")